from rest_framework.response import Response


class DataResponse(Response):
    def __init__(
        self, data=None, code=200, message="", exception=False, content_type=None, **kwargs
    ):
        response_data = {"data": data, "code": code, "message": message}
        super().__init__(
            response_data,
            exception=exception,
            content_type=content_type,
            **kwargs
        )

